package com.petrochina.cloudconfig.util;
import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.util.Enumeration;
 

import org.apache.log4j.Logger;
import org.apache.tools.zip.ZipEntry;
import org.apache.tools.zip.ZipFile;
import org.apache.tools.zip.ZipOutputStream;
 
/**
 * 
 * <pre>
 * 类名标题: 压缩工具
 * 
 * 创建人员：gl
 * 
 * 创建时间：2015年1月13日上午11:20:16
 * 
 * 说明描述: 
 * 
 * </pre>
 */
public class ZipUtil {
	public  static Logger logger=Logger.getLogger(ZipUtil.class);
 
    static final int BUFFER = 1024;
    
    /**
     * 
     * <pre>
     * 方法说明：zip
     * 
     * 开发时间：2015年1月13日上午11:10:32
     * 
     * 开发人员：gl
     * 
     * @param targetPath  D:\\ziptest\\target\\
     * @param cachePath D:\\ziptest\\cache\\TestJar\\
     * @param fileType jar/war
     * @throws Exception
     * 
     * </pre>
     *
     */
    public static void zip(String targetPath, String cachePath,String fileType)
            throws Exception {
        File inputFile = new File(cachePath);
        // 创建压缩文件
        File destDir = new File(targetPath);
        if (!destDir.exists()) {
            destDir.mkdirs();
        }
        File destFile = new File(destDir + File.separator + inputFile.getName()
                + "."+fileType);
        // 递归压缩方法
        ZipOutputStream out = new ZipOutputStream(
                new FileOutputStream(destFile));
        // 设置压缩编码.设置为utf-8后在windows下就不会乱码了，如果要放到Linux或者Unix下就不要设置了
        out.setEncoding("utf-8");
        zip(out, inputFile, "");
        logger.info("zip done");
        out.close();
    }
    /**
     * 
     * <pre>
     * 方法说明：zip core
     * 
     * 开发时间：2015年1月13日上午11:14:03
     * 
     * 开发人员：gl
     * 
     * @param out zip output stream depend on jdom
     * @param f file
     * @param base base path
     * @throws Exception 
     * 
     * </pre>
     *
     */
    public static void zip(ZipOutputStream out, File f, String base)
            throws Exception {
    	//logger.info("Zipping   " + f.getName()); // 记录日志，开始压缩
        if (f.isDirectory()) { // 如果是文件夹，则获取下面的所有文件
            File[] fl = f.listFiles();
            base = base.length() == 0 ? "" : base + "/";
            for (int i = 0; i < fl.length; i++) {
                zip(out, fl[i], base + fl[i].getName());
            }
        } else { // 如果是文件，则压缩
            out.putNextEntry(new ZipEntry(base)); // 生成下一个压缩节点
            FileInputStream in = new FileInputStream(f); // 读取文件内容
            int len;
            byte[] buf = new byte[BUFFER];
            while ((len = in.read(buf, 0, BUFFER)) != -1) {
                out.write(buf, 0, len); // 写入到压缩包
            }
            in.close();
            out.closeEntry();
        }
    }
 
    /**
     * 解压缩zip文件
     * 
     * @param sourceFile
     *            要解压的文件名 包含路径 如："D:\\ziptest\\source\\TestJar.jar"
     * @param cachePath
     *            解压后存放文件的路径 如："D:\\ziptest\\cache\\TestJar\\"
     * @throws Exception
     */
    public static void unZip(String sourceFile, String cachePath)
            throws Exception {
        ZipFile zipFile = new ZipFile(sourceFile, "utf-8"); // 以“utf-8”编码创建zip文件，用来处理winRAR压缩的文件。
        Enumeration<ZipEntry> entries = zipFile.getEntries();
        while (entries.hasMoreElements()) {
            ZipEntry entry = (ZipEntry) entries.nextElement();
            if (entry.isDirectory()) {
                File dir = new File(cachePath + entry.getName());
                if (!dir.exists()) {
                    dir.mkdirs();
                }
                continue;
            }
            BufferedInputStream bis = new BufferedInputStream(zipFile
                    .getInputStream(entry));
 
            File file = new File(cachePath + entry.getName());
            File parent = file.getParentFile();
            if (parent != null && (!parent.exists())) {
                parent.mkdirs();
            }
            FileOutputStream fos = new FileOutputStream(file);
            BufferedOutputStream bos = new BufferedOutputStream(fos, BUFFER);
            byte[] buf = new byte[BUFFER];
            int len = 0;
            while ((len = bis.read(buf, 0, BUFFER)) != -1) {
                fos.write(buf, 0, len);
            }
            bos.flush();
            bos.close();
            bis.close();
            logger.info("解压文件：" + file.getName());
        }
        zipFile.close();
    }
    
    /**
	 * 方法说明：根据路径删除指定的目录或文件，无论存在与否
	 * 开发人员：whl
     * 
	 * @param sPath
	 *            要删除的目录或文件
	 * @return 删除成功返回 true，否则返回 false。
	 */
	public static boolean DeleteFolder(String sPath) {
		boolean flag = false;
		File file = new File(sPath);
		// 判断目录或文件是否存在
		if (!file.exists()) { // 不存在返回 false
			return flag;
		} else {
			// 判断是否为文件
			if (file.isFile()) { // 为文件时调用删除文件方法
				return deleteFile(sPath);
			} else { // 为目录时调用删除目录方法
				return deleteDirectory(sPath);
			}
		}
	}

	/**
	 * 方法说明：删除单个文件
	 * 开发人员：whl
	 * @param sPath
	 *            被删除文件的文件名
	 * @return 单个文件删除成功返回true，否则返回false
	 */
	public static boolean deleteFile(String sPath) {
		boolean flag = false;
		File file = new File(sPath);
		// 路径为文件且不为空则进行删除
		if (file.isFile() && file.exists()) {
			file.delete();
			flag = true;
		}
		return flag;
	}

	/**
	 *方法说明： 删除目录（文件夹）以及目录下的文件
	 * 开发人员：whl
	 * @param sPath
	 *            被删除目录的文件路径
	 * @return 目录删除成功返回true，否则返回false
	 */
	public static boolean deleteDirectory(String sPath) {
		// 如果sPath不以文件分隔符结尾，自动添加文件分隔符
		if (!sPath.endsWith(File.separator)) {
			sPath = sPath + File.separator;
		}
		File dirFile = new File(sPath);
		// 如果dir对应的文件不存在，或者不是一个目录，则退出
		if (!dirFile.exists() || !dirFile.isDirectory()) {
			return false;
		}
		boolean flag = true;
		// 删除文件夹下的所有文件(包括子目录)
		File[] files = dirFile.listFiles();
		for (int i = 0; i < files.length; i++) {
			// 删除子文件
			if (files[i].isFile()) {
				flag = deleteFile(files[i].getAbsolutePath());
				if (!flag)
					break;
			} // 删除子目录
			else {
				flag = deleteDirectory(files[i].getAbsolutePath());
				if (!flag)
					break;
			}
		}
		if (!flag)
			return false;
		// 删除当前目录
		if (dirFile.delete()) {
			return true;
		} else {
			return false;
		}
	}
}